
from bert4keras.snippets import DataGenerator
from bert4keras.snippets import sequence_padding
from bert4keras.snippets import text_segmentate
from bert4keras.snippets import AutoRegressiveDecoder

# 基本信息
maxlen = 32
batch_size = 128
steps_per_epoch = 1000
cor